package io.renren.common.config.rabbit;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

/**
 * @ClassName RabbitMqConfig
 * @Description: TODO
 * @Author HeJie  734412450@qq.com
 * @Date 2021/1/24  21:50
 * @Version V1.0
 **/
@Configuration
public class RabbitMqConfig {

    @Resource
    RabbitTemplate rabbitTemplate;
    /**
     * 功能描述: 指定序列化方式，使用Json序列化就能在mq中看到Json数据
     *
     * @author HeJie
     * @date 2021/1/24 21:54  * @param null
     * @return null
     */
    @Bean
    public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) {
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        rabbitTemplate.setMessageConverter(new Jackson2JsonMessageConverter());
        return rabbitTemplate;
    }

    /**
     * 功能描述: 定制RabbitTemplate
     *  @PostConstruct : 表示RabbitMqConfig对象创建之后执行的方法
     *
     * @author HeJie
     * @date 2021/1/25 20:33  * @param null
     * @return null
     */
    @PostConstruct
    public void initRabbitTemplate(){
        // 设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 说明: 只要消息抵达Broker 就返回 b=true
             * @param correlationData: 当前消息的唯一关联数据（这个消息的唯一Id）
             * @param b: 消息是否成功收到
             * @param s: 若失败，则返回失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean b, String s) {

            }
        });
        // 设置消息未抵达队列的回调，即投递失败就会回调。投递成功不会调用
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 说明： 只要消息没有投递到队列就触发回调
             * @param message： 投递失败的消息详细信息
             * @param i: 回复的状态码
             * @param s： 回复的文本内容
             * @param s1 ： 消息发送给哪个交换机
             * @param s2：  消息发送用的路由键
             */
            @Override
            public void returnedMessage(Message message, int i, String s, String s1, String s2) {

            }
        });
    }
}
